package com.tx.oss;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

/**
 * 阿里云 OSS 工具类
 */
@Service
public class AliOSSUtils {

    @Autowired
    private AliOssProperties aliyunOSSProperties;

    /**
     * 实现上传图片到OSS
     */
    public String upload(MultipartFile file) throws IOException {
        // 获取上传的文件的输入流
        InputStream inputStream = file.getInputStream();

        // 避免文件覆盖
        String originalFilename = file.getOriginalFilename();
        String fileName = UUID.randomUUID().toString() + originalFilename.substring(originalFilename.lastIndexOf("."));

        //上传文件到 OSS
        OSS ossClient = new OSSClientBuilder().build(aliyunOSSProperties.getEndpoint(),
                aliyunOSSProperties.getAccessKeyId(),
                aliyunOSSProperties.getAccessKeySecret());
        ossClient.putObject(aliyunOSSProperties.getBucketName(), fileName, inputStream);

        //文件访问路径
        String url = aliyunOSSProperties.getEndpoint().split("//")[0] +
                "//" + aliyunOSSProperties.getBucketName() + "." +
                aliyunOSSProperties.getEndpoint().split("//")[1] +
                "/" + fileName;
        // 关闭ossClient
        ossClient.shutdown();
        return url;// 把上传到oss的路径返回
    }


    public List<UploadResult> batchUpload(List<MultipartFile> uploadSelfs) {
        List<UploadResult> results =new ArrayList<>();
        for (MultipartFile uploadSelf : uploadSelfs) {
            UploadResult result =new UploadResult();
            try {
                String url = upload(uploadSelf);
                result.setUrl(url);
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            result.setFilename(uploadSelf.getOriginalFilename());
            result.setObjectName(uploadSelf.getName());
            results.add(result);
        }
        return results;
    }
}
